from selenium import webdriver
import time
import unittest
import os
from public.wklogin import wklogin
from public.url import wkurl

class Tastkehu(unittest.TestCase):
    def setUp(self):
        '''setup方法，初始化，最先执行，每次都会打开网页'''
        self.driver=webdriver.Chrome()
        url = wkurl(self.driver).test_url()
        #打开悟空crm项目
        self.driver.get(url+"/index.html#/login?redirect=%2F404")
        self.driver.maximize_window()
        time.sleep(3)
        #打印执行时间
        print("start_time:"+time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime(time.time())))


    def tearDown(self):
        '''清理释放，最后执行，每次执行完都会截图，退出浏览器'''
        file_dri="D:/test/screenshot/" #指定一个文件夹
        #判断如果没有这个文件夹就创建一个用join拼接
        if not os.path.exists(file_dri):
            os.makedirs(os.path.join("D:/","test","screenshot"))
        #打印执行完时间
        print("end_time"+time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime(time.time())))
        #截图以时间戳命名（把路径加上截图时就可以直接传一个变量）
        screen_name=file_dri + time.strftime("%Y-%m-%d-%H-%M-%S",time.localtime(time.time()))+".png"
        #截图（传入含有路径和名字的变量）
        self.driver.get_screenshot_as_file(screen_name)
        self.driver.quit() #退出浏览器


    def testKehu_01(self):
        '''点击客户能否进入客户页面'''
        wklogin(self.driver).login() #调入登录方法
        self.driver.find_element_by_link_text('客户管理').click()
        time.sleep(8)
        #点击进入客户
        self.driver.find_element_by_xpath('//div[@class="container"]/ul/a[4]/li/span').click()
        time.sleep(6)
        #取得客户页面标志性控件文本
        a=self.driver.find_element_by_xpath('//*[@id="crm-table"]/div[4]/div[1]/table/thead/tr/th[3]/div/div').text
        #//*[@id="crm-table"]/div[4]/div[1]/table/thead/tr/th[3]/div/div
        #//*[@id="crm-table"]/div[2]/table/thead/tr/th[3]/div/div

        self.assertEqual("客户名称",a)#断言


    def testKehu_02(self):
        '''点击页码跳转能否成功跳转'''
        wklogin(self.driver).login()  # 调入登录方法
        self.driver.find_element_by_link_text('客户管理').click()
        time.sleep(8)
        # 点击进入客户
        self.driver.find_element_by_xpath('//div[@class="container"]/ul/a[4]/li/span').click()
        time.sleep(6)
        #拿到当前客户页面第一个客户名称
        a=self.driver.find_element_by_xpath('//div[@class="el-table__fixed"]/div[2]/table/tbody/tr[1]/td[3]/div').text
        #点击跳转页面
        self.driver.find_element_by_xpath('//button[@class="btn-next"]/i').click()
        time.sleep(3)
        #获取新页面客户名称
        b=self.driver.find_element_by_xpath('//div[@class="el-table__fixed"]/div[2]/table/tbody/tr[1]/td[3]/div').text

        self.assertNotEqual(a,b)


if __name__ == "__main__":
    unittest.main()










